"""
大模型配置
"""
from typing import Dict, Any


class ModelConfig:
    """
    大模型配置类
    """
    MODEL_TYPES = {
        "gpt": "GPT 系列模型",
        "llama": "LLaMA 系列模型",
        "bert": "BERT 系列模型"
    }

    DEFAULT_PARAMS = {
        "gpt": {"temperature": 0.7, "max_tokens": 100},
        "llama": {"temperature": 0.6, "max_tokens": 150},
        "bert": {"temperature": 0.5, "max_tokens": 200}
    }

    @classmethod
    def get_model_types(cls) -> Dict[str, str]:
        """
        获取模型类型
        """
        return cls.MODEL_TYPES

    @classmethod
    def get_default_params(cls, model_type: str) -> Dict[str, Any]:
        """
        获取默认参数
        """
        return cls.DEFAULT_PARAMS.get(model_type, {})